python program to compute the average